cd ../../../
exp_id="cdn"

backbone='resnet50'
dataset='vcoco'
dataset_path='data/v-coco'
addr="127.0.0.1"
port=1120
num_gpu=4
#pretrained='pretrained/vcoco/vcoco_cdn_s.pth'
pretrained='params/Detr/detr-r50-pre-2stage.pth'
#pretrained='logs/vcoco/hm_obj_verb_vis/stage1/checkpoint_best.pth'
#pretrained='params/ConditionalDetr/ConditionalDETR_r50-pre-2stage.pth'
#pretrained='params/DeformableDetr/r50_deformable_detr-pre-2stage.pth'
#resume='logs/vcoco/myhoi_no_hoi/stage1/checkpoint_20.pth'

export OMP_NUM_THREADS=1
python main.py \
--world_size $num_gpu \
--master_addr $addr \
--master_port $port \
--exp_id $exp_id \
--output_dir logs/$dataset/$exp_id/stage1/ \
--dataset_file $dataset \
--hoi_path $dataset_path \
--num_obj_classes 81 \
--num_verb_classes 29 \
--backbone $backbone \
--pretrained $pretrained \
--dec_layers_hopd 3 \
--dec_layers_interaction 3 \
--batch_size 4 \
--epochs 90 \
--lr_drop 60 \
--num_queries 100 \
--dim_feedforward 2048 \
--set_cost_verb_class 2 \
--verb_loss_coef 2 \
--num_feature_levels 1 \
--use_nms_filter 


